#ifndef BATCHNORM_LAYER_H
#define BATCHNORM_LAYER_H
#include "cform.h"

layer make_batchnorm_layer(int batch, int w, int h, int c);

void forward_batchnorm_layer(layer l, network net);

void backward_batchnorm_layer(layer l, network net);

#ifdef GPU

void forward_batchnorm_layer_gpu(layer l, network net);

void backward_batchnorm_layer_gpu(layer l, network net);

void pull_batchnorm_layer(layer l);

void push_batchnorm_layer(layer l);

#endif // GPU
#endif // BATCHNORM_LAYER_H